# 对数据进行分类
import ollama

# 使用模型
model= "deepseek-r1:1.5b"

# 设置输入路径
input_file = "../data/grocery_list.txt"

# 读取文件内容
with open(input_file,"r",encoding="utf-8") as f:
    items = f.read().strip()

# 给模型准备提示词 prompt
prompt = f"""
 你是一个对购物清单进行分类的助手
 以下是购物清单:
 {items}
 
 请将这些物品分类到适当的类别，例如：奶制品、蛋类、蔬菜等
"""

# 调用模型
response = ollama.generate(model=model,prompt=prompt)

result = response.get("response","")

print("模型返回：",result)






